package leetcode_jz_offer.day6_bin_tree;

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Deque;
import java.util.List;

/**
 * 二维数组形式接收，层序遍历，应用到队列，先进先出
 * 1.队列为空，层序遍历就处理完毕
 * 2.队列中保存的都是下一层要处理的元素
 * queue.size():计算出当前层的元素个数，按这个元素个数循环，将队列中当前层元素从队列中取出并放入temp集合同时将下一层元素加入队列。
 **/
public class Num32II_从上到下打印二叉树 {
    private class TreeNode {
     int val;
      TreeNode left;
     TreeNode right;
      TreeNode(int x) { val = x; }
  }
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> ret=new ArrayList<>();//定义二维数组接收形式
        if(root==null){
            return ret;
        }
        Deque<TreeNode> queue=new ArrayDeque<>();
        queue.offer(root);//根节点入队
        while (!queue.isEmpty()){
            //定义temp数组将每一层元素存入
            List<Integer> temp=new ArrayList<>();
            int size=queue.size();//当前层元素个数
            for (int i = 0; i < size; i++) {
                TreeNode cur=queue.poll();
                temp.add(cur.val);//队列中取出队首元素加入List
                if(cur.left!=null){
                    queue.offer(cur.left);
                }
                if(cur.right!=null){
                    queue.offer(cur.right);
                }
            }
            ret.add(temp);
        }
        return ret;
    }
}
